Skip to content

Conversation

CarloLucibello
Copy link
Member

@CarloLucibello CarloLucibello commented Dec 16, 2024

Porting the temporal layers to the recurrence style of Flux v0.16.

Implements the GNNRecurrence layer, similar to the next Flux.Recurrence layer but also taking the graph as input.

Also, we now export both the cells and the layer's constructor (based on GNNRecurrence) .

  • GNNRecurrence for GNNGraph and TemporalSnapshotsGNNGraph
  • GConvGRUCell and GConvGRU
  • GConvLSTMCell and GConvLSTM
  • DCGRUCell and DCGRU
  • EvolveGCNOCell
  • EvolveGCNO for temporal graph
  • A3TGCN original code and paper so poorly written that they don't deserve inclusion
  • TGCNCell and TGCN (changed, fix correctness temporalgraph implementation #548)
  • update tests
  • update docs
  • update tutorials

@CarloLucibello CarloLucibello merged commit 27d13c8 into master Dec 17, 2024
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

correctness temporalgraph implementation
1 participant